import xlrd
import os


class ReadTestData:

    @staticmethod
    def read_file_test_data(file, sheet_name):
        """
        通过excel读取测试数据
        :param file: 文件名
        :param sheet_name: 工作表sheet名
        :return: 存放每一行数据的二维列表 [[数1,数2,数3,...],[数4,数5,数6,...],...]
        """
        # 项目路径
        project_path = '\\'.join(os.path.dirname(__file__).split('\\')[:-1])
        # 测试数据文件的路径
        file_path = project_path + rf"\test_data\{file}"

        # 通过xlrd库读取excel文件
        ex = xlrd.open_workbook(file_path)
        # 根据名字指定要读取的工作区
        sheet = ex.sheet_by_name(sheet_name)
        # 将读取的数据存放到列表中
        parameter_data_list = []
        for i in range(1, sheet.nrows):
            parameter_data_list.append(sheet.row_values(i))
        return parameter_data_list


if __name__ == "__main__":
    print(ReadTestData.read_file_test_data("reading_house_test_data.xlsx", "Sheet1"))
